import dotenv
from langchain.chains.llm import LLMChain
from langchain_community.chat_models import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate

dotenv.load_dotenv()

# 构建提示词
prompt = ChatPromptTemplate.from_template("请讲一个关于{subject}主题的冷笑话")

# 构建大语言模型
llm = ChatOpenAI(model="gpt-3.5-turbo-16k")

# 构建一个chain
chain =LLMChain(prompt=prompt,llm=llm)
# 在没有LCEL表达式之前，大家很疑惑用什么表达式；目前主流的都是LCEL表达式
print(chain.invoke({'subject':'医生'}))